package com.wctj.api.base.utils;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.data.redis.core.script.DefaultScriptExecutor;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.Arrays;
import java.util.Set;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

/**
 * @author carol
 */
@Component
@Slf4j
public final class RedisUtil {

    /**
     * 用于存储标识此线程的ID
     */
    private final ThreadLocal<String> threadIdBox = new ThreadLocal<>();

    @Resource
    private RedisTemplate<String,Object> redisTemplate;

    /**
     * 加锁
     *
     * @param serviceId KEY  用于标识高并发修改资源的业务
     * @param timeout   超时时间
     * @param timeUnit  超时单位
     * @return 创建KEY成功返回true   创建KEY失败返回false
     */
    public boolean lock(String serviceId, Long timeout, TimeUnit timeUnit) {
        try {
            threadIdBox.set(UUID.randomUUID().toString());
            //如果KEY不存在，则创建KEY
            //KEY不存在，返回true  存在返回false
            return redisTemplate.opsForValue().setIfAbsent(serviceId, threadIdBox.get(), timeout, timeUnit);
        } catch (Exception e) {
            //引发任何异常，spring将不会创建KEY，直接返回false
            threadIdBox.remove();
            return false;
        }
    }


    /**
     * 释放锁
     *
     * @param serviceId KEY  用于标识高并发修改资源的业务
     * @return
     */
    public boolean unlock(String serviceId) {
        //当前线程没有生成线程标识，说明压根没有创建过KEY
        if (threadIdBox.get() == null) {
            return true;
        }
        try {
            //lua脚本(保证查询和删除的原子性)
            //如果根据serviceId查找出的值与当前线程的标识码一致，则删除该KEY并返回删除结果，没找到KEY直接返回false
            String luaScript = "if redis.call('get',KEYS[1]) == ARGV[1] then return redis.call('del',KEYS[1]) else return 0 end";
            DefaultRedisScript redisScript = new DefaultRedisScript(luaScript);
            redisScript.setResultType(Boolean.class);
            DefaultScriptExecutor defaultScriptExecutor = new DefaultScriptExecutor(redisTemplate);
            return (boolean) defaultScriptExecutor.execute(redisScript, Arrays.asList(serviceId), threadIdBox.get());
        } catch (Exception e) {
            //引发任何异常，spring将不会执行删除操作，直接返回false
            return false;
        } finally {
            threadIdBox.remove();
        }
    }


    /**
     * 指定缓存失效时间
     *
     * @param key  键
     * @param time 时间(秒)
     * @return
     */
    public boolean expire(String key, Integer time) {
        if (time > 0) {
            return redisTemplate.expire(key, time, TimeUnit.SECONDS);
        } else {
            return Boolean.FALSE;
        }
    }

    /**
     * 普通缓存获取
     *
     * @param key 键
     * @return 值
     */
    public Object get(String key) {
        return key == null ? null : redisTemplate.opsForValue().get(key);
    }

    /**
     * 删除
     *
     * @param key
     * @return
     */
    public Boolean del(String key) {
        return redisTemplate.delete(key);
    }

    /**
     * 普通缓存放入并设置时间
     *
     * @param key   键
     * @param value 值
     * @param time  时间(秒) time要大于0 如果time小于等于0 将设置无限期
     * @return true成功 false 失败
     */
    public void set(String key, Object value, long time) {
        redisTemplate.opsForValue().set(key, value, time, TimeUnit.SECONDS);
    }

    /**
     * 普通缓存放入并设置时间
     *
     * @param key   键
     * @param value 值
     * @param time  时间(秒) time要大于0 如果time小于等于0 将设置无限期
     * @return true成功 false 失败
     */
    public void set(String key, Object value, long time,TimeUnit timeUnit) {
        redisTemplate.opsForValue().set(key, value, time, timeUnit);
    }

    /**
     * @param key
     * @param value
     */
    public void set(String key, Object value) {
        redisTemplate.opsForValue().set(key, value);
    }

    /**
     * 根据key 获取过期时间
     *
     * @param key 键  不能为null
     * @return 时间(秒) 返回0代表为永久有效 -2 不存在
     */
    public long getExpire(String key) {
        return redisTemplate.getExpire(key, TimeUnit.SECONDS);
    }


    /**
     * 递增
     *
     * @param key
     * @param expire 有效时间S
     */
    public Long incr(String key, Integer expire) {
        Long count = redisTemplate.opsForValue().increment(key);
        redisTemplate.expire(key, expire, TimeUnit.SECONDS);
        return count;
    }

    public Long incrExpireWithTimeUnit(String key, Integer expire, TimeUnit timeUnit) {
        Long count = redisTemplate.opsForValue().increment(key);
        redisTemplate.expire(key, expire, timeUnit);
        return count;
    }

    /**
     * 新增不过期
     * @param key
     * @return
     */
    public Long incrNotExpire(String key) {
        Long count = redisTemplate.opsForValue().increment(key);
        return count;
    }

    /**
     * 根据前缀删除key
     * @param prefix
     */
    public void deleteByPrefix(String prefix) {
        prefix = prefix + "**";
        try {
            Set<String> keys = redisTemplate.keys(prefix);
            if (CollectionUtils.isNotEmpty(keys)) {
                redisTemplate.delete(keys);
            }
        } catch (Exception e) {
            log.error("根据前缀删除key出现异常");
        }
    }

}